tensor([[[[[0.0000, 0.6400],
           [0.0000, 0.0000]],

          [[0.0000, 0.2300],
           [0.5000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.2100, 0.4600]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.2300, 0.3600],
           [0.0000, 0.4800]],

          [[0.4400, 0.4400],
           [0.0000, 0.5400]]],


         [[[0.3200, 0.1600],
           [0.1100, 0.3100]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.2700, 0.0000],
           [0.0000, 0.5200]],

          [[0.0000, 0.3300],
           [0.2900, 0.4600]]],


         [[[0.2400, 0.3200],
           [0.4800, 0.2300]],

          [[1.0000, 1.0000],
           [1.0000, 1.0000]]]],



        [[[[0.5100, 0.0000],
           [1.0000, 0.0000]],

          [[0.5600, 0.0000],
           [0.2100, 0.0000]]],


         [[[0.4400, 0.5100],
           [0.2100, 0.0000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]]], grad_fn=<RoundBackward1>)